# -*- coding: utf-8 -*-
"""
@Author: Logan
@Date:   2025/7/7 20:35
@Description: 
"""
from time import sleep

import allure


def add_img_2_report(driver, step_name, need_sleep=True):
    """
    截图并插入allure报告
    Args:
        driver: 浏览器驱动
        step_name: 步骤名称
        need_sleep:

    Returns:

    """
    if need_sleep:
        sleep(2)
        # 核心操作：截取当前页面并附加到Allure报告
    allure.attach(
        driver.get_screenshot_as_png(),  # 截图的二进制数据（PNG格式）
        step_name + ".png",  # 截图在报告中的文件名
        allure.attachment_type.PNG  # 附件类型为PNG图片
    )

def add_img_path_2_report(img_path, step_name):
    """
    将本地图片插入allure报告（如验证码）
    Args:
        img_path: 图片路径
        step_name: 步骤名称
    Returns:

    """
    allure.attach.file(img_path, step_name, allure.attachment_type.PNG)
